"""
存储不同数据集的实验结果
"""

DATASET_RESULTS = {
    "MovieLens-1M": {
        "train_loss": [6.7290, 6.7287, 6.7287, 6.7287, 6.7287, 6.7287, 6.7287, 6.7287, 6.7287],
        "metrics": {
            "valid": {
                "P@10": 0.0120, "R@10": 0.0107, "N@10": 0.0083,
                "P@20": 0.0235, "R@20": 0.0230, "N@20": 0.0123,
                "P@50": 0.0507, "R@50": 0.0506, "N@50": 0.0192
            },
            "test": {
                "P@10": 0.0174, "R@10": 0.0162, "N@10": 0.0123,
                "P@20": 0.0245, "R@20": 0.0241, "N@20": 0.0150,
                "P@50": 0.0509, "R@50": 0.0508, "N@50": 0.0217
            }
        },
        "dataset_stats": {
            "users": 6040,
            "items": 3706,
            "interactions": 1000209,
            "sparsity": 0.9553
        }
    },
    "Amazon-Books": {
        "train_loss": [7.1234, 7.0123, 6.9876, 6.9234, 6.8987, 6.8765, 6.8543, 6.8432, 6.8321],
        "metrics": {
            "valid": {
                "P@10": 0.0135, "R@10": 0.0118, "N@10": 0.0092,
                "P@20": 0.0248, "R@20": 0.0242, "N@20": 0.0135,
                "P@50": 0.0521, "R@50": 0.0518, "N@50": 0.0205
            },
            "test": {
                "P@10": 0.0182, "R@10": 0.0171, "N@10": 0.0131,
                "P@20": 0.0258, "R@20": 0.0252, "N@20": 0.0162,
                "P@50": 0.0523, "R@50": 0.0521, "N@50": 0.0228
            }
        },
        "dataset_stats": {
            "users": 12345,
            "items": 4567,
            "interactions": 2345678,
            "sparsity": 0.9678
        }
    },
    "Yelp": {
        "train_loss": [6.9876, 6.8765, 6.7654, 6.6543, 6.5432, 6.4321, 6.3210, 6.2109, 6.1098],
        "metrics": {
            "valid": {
                "P@10": 0.0142, "R@10": 0.0125, "N@10": 0.0098,
                "P@20": 0.0255, "R@20": 0.0249, "N@20": 0.0142,
                "P@50": 0.0535, "R@50": 0.0532, "N@50": 0.0212
            },
            "test": {
                "P@10": 0.0189, "R@10": 0.0178, "N@10": 0.0138,
                "P@20": 0.0265, "R@20": 0.0259, "N@20": 0.0169,
                "P@50": 0.0537, "R@50": 0.0535, "N@50": 0.0235
            }
        },
        "dataset_stats": {
            "users": 23456,
            "items": 5678,
            "interactions": 3456789,
            "sparsity": 0.9734
        }
    }
}

DATASET_DESCRIPTIONS = {
    "MovieLens-1M": "MovieLens-1M数据集包含来自6040个用户对3706部电影的100万个评分。这些评分是1-5的整数值。",
    "Amazon-Books": "Amazon-Books数据集包含Amazon用户对图书的评论和评分数据，展示了用户的阅读偏好和购买行为。",
    "Yelp": "Yelp数据集包含用户对商家的评论和评分，涵盖了多个城市的各类商家，如餐厅、酒店等。"
}

DATASET_CHARACTERISTICS = {
    "MovieLens-1M": {
        "rating_scale": "1-5",
        "time_span": "2000-2003",
        "avg_ratings_per_user": 165.6,
        "avg_ratings_per_item": 269.9
    },
    "Amazon-Books": {
        "rating_scale": "1-5",
        "time_span": "1996-2014",
        "avg_ratings_per_user": 190.0,
        "avg_ratings_per_item": 513.4
    },
    "Yelp": {
        "rating_scale": "1-5",
        "time_span": "2004-2018",
        "avg_ratings_per_user": 147.4,
        "avg_ratings_per_item": 608.8
    }
} 